import torch


class EnergyConfig:
    n_features = 6
    seq_length = 5
    hidden_size = 256
    num_layers = 1
    num_classes = 1
    batch_size = 5
    learning_rate = 0.005
    num_epochs = 2000
    dropout_prob = 0.3
    device = torch.device('mps')
